package com.example.healthmonitoringbe;


import com.alibaba.fastjson.JSONObject;

import com.example.entity.*;
import com.example.service.*;
import org.apache.commons.codec.binary.Base64;
import org.apache.qpid.jms.JmsConnection;
import org.apache.qpid.jms.JmsConnectionListener;
import org.apache.qpid.jms.message.JmsInboundMessageDispatch;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Configuration;

import javax.crypto.Mac;
import javax.crypto.spec.SecretKeySpec;
import javax.jms.*;
import javax.naming.Context;
import javax.naming.InitialContext;
import java.net.URI;
import java.time.LocalDateTime;
import java.util.Hashtable;
import java.util.List;

@Configuration
public class DeviceMessage implements InitializingBean {

    @Value("${user.accessKeyId}")
    private String accessKeyId;

    @Value("${user.accessKeySecret}")
    private String accessKeySecret;

    @Value("${user.uid}")
    private String uid;

    @Value("${iot.regionId}")
    private String regionId;

    @Value("${iot.instance}")
    private String instance;

    private final IDeviceInfoService deviceInfoService;
    private final IBgDeviceHistoryLogService bgDeviceHistoryLogService;
    private final IBpDeviceHistoryLogService bpDeviceHistoryLogService;
    private final IBpAlarmHistoryLogService bpAlarmHistoryLogService;
    private final IBgAlarmHistoryLogService bgAlarmHistoryLogService;

    public DeviceMessage(IDeviceInfoService deviceInfoService, IBgDeviceHistoryLogService bgDeviceHistoryLogService, IBpDeviceHistoryLogService bpDeviceHistoryLogService, IBpAlarmHistoryLogService bpAlarmHistoryLogService, IBgAlarmHistoryLogService bgAlarmHistoryLogService) {
        this.deviceInfoService = deviceInfoService;
        this.bgDeviceHistoryLogService = bgDeviceHistoryLogService;
        this.bpDeviceHistoryLogService = bpDeviceHistoryLogService;
        this.bpAlarmHistoryLogService = bpAlarmHistoryLogService;
        this.bgAlarmHistoryLogService = bgAlarmHistoryLogService;
    }

    @Override
    public void afterPropertiesSet() throws Exception {

        long timeStamp = System.currentTimeMillis();
        //签名方法：支持hmacmd5、hmacsha1和hmacsha256。
        String signMethod = "hmacsha1";

        //userName组装方法，请参见AMQP客户端接入说明文档。
        String userName = "ac:de:48:00:11:22" + "-" + 0 + "|authMode=aksign"
                + ",signMethod=" + signMethod
                + ",timestamp=" + timeStamp
                + ",authId=" + accessKeyId
                + ",iotInstanceId=" + instance
                + ",consumerGroupId=" + "DEFAULT_GROUP"
                + "|";
        //计算签名，password组装方法，请参见AMQP客户端接入说明文档。
        String signContent = "authId=" + accessKeyId + "&timestamp=" + timeStamp;
        String password = doSign(signContent, accessKeySecret, signMethod);
        String host = uid + ".iot-amqp." + regionId + ".aliyuncs.com";
        String connectionUrl = "failover:(amqps://" + host + ":5671?amqp.idleTimeout=80000)"
                + "?failover.reconnectDelay=30";

        Hashtable<String, String> hashtable = new Hashtable<>();
        hashtable.put("connectionfactory.SBCF", connectionUrl);
        hashtable.put("queue.QUEUE", "default");
        hashtable.put(Context.INITIAL_CONTEXT_FACTORY, "org.apache.qpid.jms.jndi.JmsInitialContextFactory");
        Context context = new InitialContext(hashtable);
        ConnectionFactory cf = (ConnectionFactory) context.lookup("SBCF");
        Destination queue = (Destination) context.lookup("QUEUE");
        // 创建连接。
        Connection connection = cf.createConnection(userName, password);
//        connections.add(connection);

        ((JmsConnection) connection).addConnectionListener(myJmsConnectionListener);
        // 创建会话。
        // Session.CLIENT_ACKNOWLEDGE: 收到消息后，需要手动调用message.acknowledge()。
        // Session.AUTO_ACKNOWLEDGE: SDK自动ACK（推荐）。
        Session session = connection.createSession(false, Session.AUTO_ACKNOWLEDGE);

        connection.start();
        MessageListener messageListener = message -> {
            try {
                //1.收到消息之后一定要ACK。
                // 推荐做法：创建Session选择Session.AUTO_ACKNOWLEDGE，这里会自动ACK。
                // 其他做法：创建Session选择Session.CLIENT_ACKNOWLEDGE，这里一定要调message.acknowledge()来ACK。
                // message.acknowledge();
                //2.建议异步处理收到的消息，确保onMessage函数里没有耗时逻辑。
                // 如果业务处理耗时过程过长阻塞住线程，可能会影响SDK收到消息后的正常回调。
                processMessage(message);
            } catch (Exception e) {
                System.out.println("submit task occurs exception " + e);
            }
        };

        // 创建Receiver连接。
        MessageConsumer consumer = session.createConsumer(queue);
        consumer.setMessageListener(messageListener);

    }

    /**
     * 在这里处理您收到消息后的具体业务逻辑。
     */
    private void processMessage(Message message) {
        try {
            byte[] body = message.getBody(byte[].class);
            String content = new String(body);
            String topic = message.getStringProperty("topic");
            String messageId = message.getStringProperty("messageId");
            System.out.println("receive message"
                    + ",\n topic = " + topic
                    + ",\n messageId = " + messageId
                    + ",\n content = " + content);
            // =========================================================================== //
            if (topic.contains("/as/mqtt/status/")) {
                JSONObject jsonObject = JSONObject.parseObject(content);
                DeviceInfo deviceInfo = new DeviceInfo();
                deviceInfo.setDeviceName(jsonObject.getString("deviceName"));
                deviceInfo.setConnectStatus(jsonObject.getString("status").equals("online"));
                if (jsonObject.getString("status").equals("online")) {
                    deviceInfo.setLastOnlineDate(LocalDateTime.now());
                }
                deviceInfo.setGmtUpdate(LocalDateTime.now());
                deviceInfoService.saveDeviceInFo(deviceInfo);
            } else if (topic.contains("/user/healthMonitoringUpLoad")) {
                // 设定健康监测数据 基准值 正常血压、正常血糖的范围
                JSONObject jsonObject = JSONObject.parseObject(content);
                DeviceInfo deviceInfo = new DeviceInfo();
                deviceInfo.setDeviceName(jsonObject.getString("deviceName"));
                deviceInfo.setBorderBpUpMax(jsonObject.getFloatValue("border_bp_up_max"));
                deviceInfo.setBorderBpUpMin(jsonObject.getFloat("border_bp_up_min"));
                deviceInfo.setBorderBpDownMax(jsonObject.getFloat("border_bp_down_max"));
                deviceInfo.setBorderBpDownMin(jsonObject.getFloat("border_bp_down_min"));
                deviceInfo.setBorderBgMax(jsonObject.getFloat("border_bg_max"));
                deviceInfo.setBorderBgMin(jsonObject.getFloat("border_bg_min"));
                deviceInfo.setGmtUpdate(LocalDateTime.now());
                deviceInfoService.saveDeviceInFo(deviceInfo);
            } else if (topic.contains("/user/currentBpUpload")) {
                // 当前血压数据上传
                JSONObject jsonObject = JSONObject.parseObject(content);

                DeviceInfo deviceInfo = new DeviceInfo();
                deviceInfo.setDeviceName(jsonObject.getString("deviceName"));
                deviceInfo.setCurrentBpUp(jsonObject.getFloat("current_bp_up"));
                deviceInfo.setCurrentBpDown(jsonObject.getFloat("current_bp_down"));
                // deviceInfoService.saveDeviceInFo(deviceInfo); // 这条语句放在后面 方便在警报时记录警报时间

                BpDeviceHistoryLog bpDeviceHistoryLog = new BpDeviceHistoryLog();
                bpDeviceHistoryLog.setDeviceName(jsonObject.getString("deviceName"));
                bpDeviceHistoryLog.setCurrentBpUp(jsonObject.getFloat("current_bp_up"));
                bpDeviceHistoryLog.setCurrentBpDown(jsonObject.getFloat("current_bp_down"));
                bpDeviceHistoryLog.setGmtCreate(LocalDateTime.now());
                bpDeviceHistoryLogService.save(bpDeviceHistoryLog);

                // 查找当前设备的标注值
                DeviceInfo currentDeviceInfo = deviceInfoService.findDeviceByName(jsonObject.getString("deviceName"));
                // 查找血压警报数据 计入历史数据的id
                List<BpDeviceHistoryLog> bpDeviceHistoryLogs = bpDeviceHistoryLogService.list();
                String bpDeviceId = bpDeviceHistoryLogs.get(bpDeviceHistoryLogs.size() - 1).getId();
                // 对数据进行判断，如果异常，则需要记录异常数据。
                if (!(jsonObject.getFloat("current_bp_up") <= currentDeviceInfo.getBorderBpUpMax()
                        && jsonObject.getFloat("current_bp_up") >= currentDeviceInfo.getBorderBpUpMin())
                        || !(jsonObject.getFloat("current_bp_down") <= currentDeviceInfo.getBorderBpDownMax()
                        && jsonObject.getFloat("current_bp_down") >= currentDeviceInfo.getBorderBpDownMin())) {
                    BpAlarmHistoryLog bpAlarmHistoryLog = new BpAlarmHistoryLog();
                    bpAlarmHistoryLog.setDeviceName(jsonObject.getString("deviceName"));
                    bpAlarmHistoryLog.setAlarmBpUp(jsonObject.getFloat("current_bp_up"));
                    bpAlarmHistoryLog.setAlarmBpDown(jsonObject.getFloat("current_bp_down"));
                    bpAlarmHistoryLog.setGmtCreate(LocalDateTime.now());
                    bpAlarmHistoryLog.setBpDeviceId(bpDeviceId);
                    bpAlarmHistoryLogService.save(bpAlarmHistoryLog);
                    // 记录警报时间
                    deviceInfo.setLastAlarmDate(LocalDateTime.now());
                }

                deviceInfoService.saveDeviceInFo(deviceInfo);
            } else if (topic.contains("/user/currentBgUpload")) {
                // 当前血压、血糖数据上传
                JSONObject jsonObject = JSONObject.parseObject(content);

                DeviceInfo deviceInfo = new DeviceInfo();
                deviceInfo.setDeviceName(jsonObject.getString("deviceName"));
                deviceInfo.setCurrentBg(jsonObject.getFloat("current_bg"));
                // deviceInfoService.saveDeviceInFo(deviceInfo); // 这条语句放在后面 方便在警报时记录警报时间

                BgDeviceHistoryLog bgDeviceHistoryLog = new BgDeviceHistoryLog();
                bgDeviceHistoryLog.setDeviceName(jsonObject.getString("deviceName"));
                bgDeviceHistoryLog.setCurrentBg(jsonObject.getFloat("current_bg"));
                bgDeviceHistoryLog.setGmtCreate(LocalDateTime.now());
                bgDeviceHistoryLogService.save(bgDeviceHistoryLog);

                // 查找当前设备的标注值
                DeviceInfo currentDeviceInfo = deviceInfoService.findDeviceByName(jsonObject.getString("deviceName"));
                // 查找血糖警报数据 计入历史数据的id
                List<BgDeviceHistoryLog> bgDeviceHistoryLogs = bgDeviceHistoryLogService.list();
                String bgDeviceId = bgDeviceHistoryLogs.get(bgDeviceHistoryLogs.size() - 1).getId();
                if (!(jsonObject.getFloat("current_bg") <= currentDeviceInfo.getBorderBgMax() &&
                        jsonObject.getFloat("current_bg") >= currentDeviceInfo.getBorderBgMin())) {
                    BgAlarmHistoryLog bgAlarmHistoryLog = new BgAlarmHistoryLog();
                    bgAlarmHistoryLog.setDeviceName(jsonObject.getString("deviceName"));
                    bgAlarmHistoryLog.setAlarmBg(jsonObject.getFloat("current_bg"));
                    bgAlarmHistoryLog.setGmtCreate(LocalDateTime.now());
                    bgAlarmHistoryLog.setBgDeviceId(bgDeviceId);
                    bgAlarmHistoryLogService.save(bgAlarmHistoryLog);
                    // 记录警报时间
                    deviceInfo.setLastAlarmDate(LocalDateTime.now());
                }

                deviceInfoService.saveDeviceInFo(deviceInfo);
            }
            // =========================================================================== //
        } catch (Exception e) {
            System.out.println("processMessage occurs error " + e);
        }
    }


    /**
     * 计算签名，password组装方法，请参见AMQP客户端接入说明文档。
     */
    private static String doSign(String toSignString, String secret, String signMethod) throws Exception {
        SecretKeySpec signingKey = new SecretKeySpec(secret.getBytes(), signMethod);
        Mac mac = Mac.getInstance(signMethod);
        mac.init(signingKey);
        byte[] rawHmac = mac.doFinal(toSignString.getBytes());
        return Base64.encodeBase64String(rawHmac);
    }

    private static final JmsConnectionListener myJmsConnectionListener = new JmsConnectionListener() {
        /**
         * 连接成功建立。
         */
        @Override
        public void onConnectionEstablished(URI remoteURI) {
            System.out.println("onConnectionEstablished, remoteUri:{}" + remoteURI);
        }

        /**
         * 尝试过最大重试次数之后，最终连接失败。
         */
        @Override
        public void onConnectionFailure(Throwable error) {
            System.out.println("onConnectionFailure, {}" + error.getMessage());
        }

        /**
         * 连接中断。
         */
        @Override
        public void onConnectionInterrupted(URI remoteURI) {
            System.out.println("onConnectionInterrupted, remoteUri:{}" + remoteURI);
        }

        /**
         * 连接中断后又自动重连上。
         */
        @Override
        public void onConnectionRestored(URI remoteURI) {
            System.out.println("onConnectionRestored, remoteUri:{}" + remoteURI);
        }

        @Override
        public void onInboundMessage(JmsInboundMessageDispatch envelope) {
        }

        @Override
        public void onSessionClosed(Session session, Throwable cause) {
        }

        @Override
        public void onConsumerClosed(MessageConsumer consumer, Throwable cause) {
        }

        @Override
        public void onProducerClosed(MessageProducer producer, Throwable cause) {
        }
    };
}
